from appium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import os,time
from time import sleep

class Base():
    def __init__(self):
        os.system('abc root')
        cap = {
            "platformName": "Android",
            "deviceName": "emulator-5554",
            "appPackage": "cn.xuexi.android",
            "appActivity": "com.alibaba.android.rimet.biz.home.activity.HomeActivity",
            "noReset": True
        }
        self.driver = webdriver.Remote('http://127.0.0.1:4327/wd/hub',cap)
        self.driver.implicitly_wait(25)
        sleep(15)

    def find_ele(self,loc):
        try:
            if type(loc) == str:
                return self.driver.find_element_by_android_uiautomator(loc)
            else:
                return self.driver.find_element(*loc)
        except Exception as e :
            print("报错信息",e)
            print("未找到页面元素",loc)

    def find_wait_ele(self, loc):
        try:
            return WebDriverWait(self.driver,15).until(EC.presence_of_element_located(loc))
        except Exception as e:
            print("报错信息", e)
            print("未找到页面元素", loc)

    def jieping(self,filename):
        t = time.strftime('%Y%m%d%H%M%S',time.localtime())
        self.driver.save_screenshot(t+filename)

    def get_key_value(self,loc,key):
        ele = self.find_ele(loc)
        return ele.get_attribute(key)

    def quit_driver(self):
        self.driver.quit()